package map_and_set;

import java.util.HashMap;

class Node {
    int val;
    Node next;
    Node random;

    public Node(int val) {
        this.val = val;
        this.next = null;
        this.random = null;
    }
}
public class CopyRandomList {
    public Node copyRandomList(Node head) {
        // 创建一个 map key值存放旧节点 val值存放新节点
        HashMap<Node,Node> map = new HashMap<>();
        Node cur = head;
        // 遍历原来的列表 将原来的列表节点进行复制 只复制 val 。next 和 random下面进行复制
        while(cur != null){
            Node node = new Node(cur.val);
            map.put(cur,node);
            cur = cur.next;
        }
        cur = head;
        // 开始进行拷贝 next 和 random
        while(cur != null){
            // get方法 取到cur对应的val值 val就是新节点 让新节点的next 等于 让原来节点cur的next 对应的val
            map.get(cur).next = map.get(cur.next);
            // 同上
            map.get(cur).random = map.get(cur.random);
            cur = cur.next;
        }
        return map.get(head);
    }
}
